步驟 3:性能評估
1.通過 PSNR 和 SSIM 指標對修復效果進行量化評估。
2.進行定性分析,觀察填補區域的無縫性與自然度。
from skimage.metrics import peak_signal_noise_ratio, structural_similarity
計算 PSNR
psnr_value = peak_signal_noise_ratio(image, output_image)
print(f'PSNR: {psnr_value}')
計算 SSIM
ssim_value = structural_similarity(image, output_image, multichannel=True)
print(f'SSIM: {ssim_value}')
實驗結果
• 簡單遮擋測試結果:
• PSNR:40.56
• SSIM:0.95
• 定性分析:填補區域與周圍背景無縫融合,修復效果良好,無明顯異常。
• 複雜遮擋測試結果:
• PSNR:28.34
• SSIM:0.82
• 定性分析:複雜背景與物件重疊場景下,修復區域在某些細節處顯得不夠自然,出現少量的異常紋理或邊界模糊情況。
結論
• DeepFill v2 在簡單遮擋場景下表現優異,能夠無縫修復圖像。然而,當遇到複雜背景或物件重疊的遮擋情況時,模型的表現有所下降,修復區域的自然度和無縫性有所減弱。通過 PSNR 和 SSIM 的定量分析,發現模型在複雜場景下的修復效果仍有提升空間。